package cn.edu.nyist.util;

import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author mumucgq
 * @create 2020-03-16 15:48
 */
public class ExcelUtils {

    private static Logger log = LoggerFactory.getLogger(ExcelUtils.class);

    public static boolean isExcel2003=false;
    /*
    * @描述：是否是2003的excel，返回true是2003
    * */
    public static boolean isExcel2003(String filePath)  {
         boolean matches = filePath.matches("^.+\\.(?i)(xls)$");
        log.info("isExcel2003"+matches);
         if (matches){
             isExcel2003=true;
         }
        return matches;
    }

    /*
    * 描述：是否是2007的excel，返回true是2007
    * */
    public static boolean isExcel2007(String filePath)  {
        boolean matches = filePath.matches("^.+\\.(?i)(xlsx)$");
        log.info("isExcel2007"+matches);
        if (matches){
            isExcel2003=false;
        }
        return matches;
    }

    /*
    * 验证是否是EXCEL文件
    * */
    public static boolean validateExcel(String filePath){
        isExcel2003(filePath);
        isExcel2007(filePath);
        if (filePath == null || isExcel2003(filePath) || isExcel2007(filePath)){
            return false;
        }
        return true;
    }
    /**
     * 导出Excel
     * @param sheetName sheet名称
     * @param title 标题
     * @param values 内容
     * @param wb HSSFWorkbook对象
     * @return
     */
    public static HSSFWorkbook getHSSFWorkbook(String sheetName, String []title, String [][]values, HSSFWorkbook wb){
        // 第一步，创建一个HSSFWorkbook，对应一个Excel文件
        if(wb == null){
            wb = new HSSFWorkbook();
        }
        // 第二步，在workbook中添加一个sheet,对应Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet(sheetName);
        // 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制
        HSSFRow row = sheet.createRow(0);
        // 第四步，创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HorizontalAlignment.CENTER); // 创建一个居中格式
        //声明列对象
        HSSFCell cell = null;
        //创建标题
        for(int i=0;i<title.length;i++){
            cell = row.createCell(i);
            cell.setCellValue(title[i]);
            cell.setCellStyle(style);
        }
        //创建内容
        for(int i=0;i<values.length;i++){
            row = sheet.createRow(i + 1);
            for(int j=0;j<values[i].length;j++){
                //将内容按顺序赋给对应的列对象
                row.createCell(j).setCellValue(values[i][j]);
            }
        }
        return wb;
    }

}
